package com.sheng.leetcode.year2022.month05.day06;

import org.junit.Test;

import java.util.HashMap;
import java.util.Map;

/**
 * @author liusheng
 * @date 2022/05/06
 *
 * 1387. 将整数按权重排序
 *
 * 我们将整数 x 的 权重 定义为按照下述规则将 x 变成 1所需要的步数：
 *
 * 如果 x 是偶数，那么 x = x / 2
 * 如果 x 是奇数，那么 x = 3 * x + 1
 * 比方说，x=3 的权重为 7 。
 * 因为 3 需要 7 步变成 1 （3 --> 10 --> 5 --> 16 --> 8 --> 4 --> 2 --> 1）。
 *
 * 给你三个整数 lo， hi 和 k 。
 * 你的任务是将区间 [lo, hi] 之间的整数按照它们的权重 升序排序 ，
 * 如果大于等于 2 个整数有 相同 的权重，那么按照数字自身的数值 升序排序 。
 * 请你返回区间 [lo, hi] 之间的整数按权重排序后的第 k 个数。
 * 注意，题目保证对于任意整数 x （lo <= x <= hi） ，它变成 1 所需要的步数是一个 32 位有符号整数。
 *
 * 示例 1：
 *
 * 输入：lo = 12, hi = 15, k = 2
 * 输出：13
 * 解释：12 的权重为 9（12 --> 6 --> 3 --> 10 --> 5 --> 16 --> 8 --> 4 --> 2 --> 1）
 * 13 的权重为 9
 * 14 的权重为 17
 * 15 的权重为 17
 * 区间内的数按权重排序以后的结果为 [12,13,14,15] 。对于 k = 2 ，答案是第二个整数也就是 13 。
 * 注意，12 和 13 有相同的权重，所以我们按照它们本身升序排序。14 和 15 同理。
 * 示例 2：
 *
 * 输入：lo = 7, hi = 11, k = 4
 * 输出：7
 * 解释：区间内整数 [7, 8, 9, 10, 11] 对应的权重为 [16, 3, 19, 6, 14] 。
 * 按权重排序后得到的结果为 [8, 10, 11, 7, 9] 。
 * 排序后数组中第 4 个数字为 7 。
 *
 * 提示：
 *
 * 1 <= lo <= hi <= 1000
 * 1 <= k <= hi - lo + 1
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/sort-integers-by-the-power-value
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class LeetCode1387 {

    @Test
    public void test01(){
        //int lo = 12, hi = 15, k = 2;
        int lo = 7, hi = 11, k = 4;
        System.out.println(new Solution().getKth(lo, hi, k));
    }

}
class Solution {
    public int getKth(int lo, int hi, int k) {
        int[] ints = new int[hi - lo + 1];
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = lo; i <= hi; i++) {
            int num = i;
            int count = -1;
            while (num != 0) {
                num = dfs(num);
                count++;
            }
            map.put(i, count);
            ints[i - lo] = i;
        }
        for (int i = 0; i < ints.length - 1; i++) {
            for (int j = i + 1; j < ints.length; j++) {
                if (map.get(ints[i]) > map.get(ints[j])) {
                    int temp = ints[i];
                    ints[i] = ints[j];
                    ints[j] = temp;
                } else if (map.get(ints[i]).equals(map.get(ints[j]))) {
                    if (ints[i] > ints[j]) {
                        int temp = ints[i];
                        ints[i] = ints[j];
                        ints[j] = temp;
                    }
                }
            }
        }
        return ints[k - 1];
    }

    public int dfs(int num) {
        if (num == 1) {
            return 0;
        }
        return num % 2 == 0 ? num / 2 : 3 * num + 1;
    }
}
